from paddlers.deploy import Predictor


class Change:
    def __init__(self):
        self.predictor = Predictor("./model/change", use_gpu=True)

    def forward(self, img1, img2):
        res = self.predictor.predict((img1, img2))
        return res[0]['label_map']*255